<template>
  <el-card>
    <div class="head-container">
      <el-input
        v-model="searchText"
        placeholder="请输入部门名称"
        clearable
        size="small"
        prefix-icon="el-icon-search"
        style="margin-bottom: 20px"
      />
    </div>
    <div>
      <el-radio-group v-model="seltype" size="small">
        <el-radio-button label="river">水源地类型</el-radio-button>
        <el-radio-button label="region">行政区划</el-radio-button>
      </el-radio-group>
    </div>
    <div class="head-container">
      <el-tree
        :data="deptOptions"
        :props="defaultProps"
        :expand-on-click-node="false"
        :filter-node-method="filterNode"
        show-checkbox
        default-expand-all
        ref="tree"
        node-key="id"
        highlight-current
        :default-checked-keys="defaultCheckedKeys"
        @node-click="handleNodeClick"
        @check-change="handleCheckChange"
        @check="handleCheck"
      />
    </div>
  </el-card>
</template>

<script>
import { getLakeSegmentTree, getRegionTree } from "@/api/dataResources/common";
import { getRiverSegmentTree } from "@/api/commonApi";
export default {
  name: "TreeSelect",
  data() {
    return {
      searchText: "",
      seltype: "river",
      deptOptions: [],
      defaultProps: {
        children: "children",
        label: "label",
      },
      defaultCheckedKeys: [], // 默认选中的节点ID数组
      allNodes: [], // 存储所有节点，用于判断是否为子节点
    };
  },
  watch: {
    searchText(val) {
      this.$refs.tree.filter(val);
    },
    seltype: {
      handler(newVal) {
        if (newVal === "river") {
          this.getRiverTree();
        } else if (newVal === "region") {
          this.getRegionTree();
        }
      },
      immediate: true, // 初始化时也会执行
    },
  },
  methods: {
    // 节点过滤方法
    filterNode(value, data) {
      if (!value) return true;
      return data.label.indexOf(value) !== -1;
    },

    // 收集所有节点（用于判断是否为子节点）
    collectAllNodes(data) {
      const nodes = [];
      const traverse = (tree) => {
        tree.forEach((node) => {
          nodes.push(node);
          if (node.children && node.children.length) {
            traverse(node.children);
          }
        });
      };
      traverse(data);
      return nodes;
    },

    // 判断是否为子节点（有子节点的就是父节点，不需要传递）
    isLeafNode(nodeId) {
      const node = this.allNodes.find((n) => n.id === nodeId);
      return node && (!node.children || node.children.length === 0);
    },

    // 将扁平数据转换为树形结构
    convertToTree(data) {
      // 创建id到节点的映射表
      const nodeMap = {};

      // 初始化所有节点，保留原有id，添加label和children
      data.forEach((node) => {
        nodeMap[node.id] = {
          ...node, // 保留所有原始字段（包括id）
          label: node.name, // 添加label字段，值为name
          children: [], // 初始化子节点数组
        };
      });

      // 构建树形结构
      const tree = [];
      data.forEach((node) => {
        const currentNode = nodeMap[node.id];
        const parentNode = nodeMap[node.pId];

        if (parentNode) {
          // 存在父节点，添加到父节点的children中
          parentNode.children.push(currentNode);
        } else {
          // 根节点（pId为0）直接加入树
          tree.push(currentNode);
        }
      });

      return tree;
    },

    // 设置默认选中所有节点
    setDefaultChecked() {
      // 收集所有节点
      this.allNodes = this.collectAllNodes(this.deptOptions);

      // 获取所有节点ID，设置为默认选中
      this.defaultCheckedKeys = this.allNodes.map((node) => node.id);

      // 如果树已渲染，强制更新选中状态
      if (this.$refs.tree) {
        this.$refs.tree.setCheckedKeys(this.defaultCheckedKeys);
      }

      // 初始就触发一次选中事件，将子节点ID传递给父组件
      this.handleCheck(null, { checkedNodes: this.allNodes });
    },

    // 获取区划
    getRegionTree() {
      let query = {
        segmentType: "IsAuto",
        segmentTypeDrink: "filter",
      };
      getRegionTree(query).then((res) => {
        if (res.code === 200) {
          this.deptOptions = this.convertToTree(res.data || []);
          this.setDefaultChecked(); // 数据加载完成后设置默认全选
        }
      });
    },

    // 获取河流树
    getRiverTree() {
      let query = {
        segmentType: "IsAuto",
        segmentTypeDrink: "filter",
      };
      getRiverSegmentTree(query).then((res) => {
        if (res.code === 200) {
          this.deptOptions = this.convertToTree(res.data || []);
          this.setDefaultChecked(); // 数据加载完成后设置默认全选
        }
      });
    },

    // 点击节点时触发
    handleNodeClick(data, node) {
      // 点击事件处理
    },

    // 节点勾选状态变化时触发
    handleCheckChange(data, checked, indeterminate) {
      // 勾选状态变化处理
    },

    // 节点勾选事件（包含全选/半选状态）
    handleCheck(selection, checkedKeys) {
      // 筛选出所有子节点（非父节点）的ID
      const leafNodeIds = checkedKeys.checkedNodes
        .filter((node) => this.isLeafNode(node.id))
        .map((node) => node.id);

      // 传递给父组件
      this.$emit("node-click", leafNodeIds.join(","));
    },
  },
};
</script>

<style></style>
